leetcode arrays problem with solution